import os

class Config:
    def __init__(self):
        self.project_dir = os.path.dirname(os.path.abspath(__file__))
        self.data_path = os.path.join(self.project_dir, 'data')
        self.model_path = os.path.join(self.project_dir, 'model')
        self.pretrained_LM_path = os.path.join(self.project_dir, 'pretrained-LM')
        self.result_path = os.path.join(self.project_dir, 'result')
        self.task_path = os.path.join(self.project_dir, 'task')
        self.tools_path = os.path.join(self.project_dir, 'tools')

        # data
        self.knowledge_bases_path = os.path.join(self.data_path, 'knowledge-bases')
        self.multiple_choices_path = os.path.join(self.data_path, 'multiple_choices')
        self.video_courses_path = os.path.join(self.data_path, 'video-courses')


        # pretrained-lm
        self.bert_base_chinese_path = os.path.join(self.pretrained_LM_path, 'bert-base-chinese')
        self.bert_base_uncased_path = os.path.join(self.pretrained_LM_path, 'bert-base-uncased')
        self.bert_with_text_path = os.path.join(self.pretrained_LM_path,'pretrained-with-text')
        self.bert_with_video_path = os.path.join(self.pretrained_LM_path,'bert_with_video')


        # result
        self.generated_text_path = os.path.join(self.result, 'generated-text')
        self.multiple_choices_QA_NSP_result_path = os.path.join(self.result, 'multiple-choice-QA-result_NSP')
        self.multiple_choices_QA_perplexity_result_path = os.path.join(self.result, 'multiple-choice-QA-result_perplexity')

if __name__ == '__main__':
    config = Config()
    print(os.path.exists(config.bert_base_chinese_path))